"""
张量的索引：
[   第一列   第二列  第三列
    索引0    索引1   索引2
    [1,      2,     3]    第一行 索引 0
    [3,      4,     5]    第二行 索引 1
    [5,      6,     7]    第二行 索引 2
]
行列索引：
语法：data[行, 列]
"""
import torch as t

data = t.randint(1, 5, [3, 3])
print("data:", data)
print("data的形状:", data.size())

# 获取行列索引
print("data[0,1]:", data[0, 1])
print("data[1,2]:", data[1, 2])

# 获取第一行
print("获取第一行方式一:", data[0])
print("获取第一行方式二:", data[0, :])

# 获取第一列
print("获取第一列:", data[:, 0])
